package com.guiji.quartz.domain;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.crypto.SecureUtil;
import cn.hutool.http.HttpException;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.http.HttpUtil;
import cn.hutool.json.JSONUtil;
import cn.wanghaomiao.xpath.model.JXDocument;
import com.guiji.common.utils.spring.SpringUtils;
import com.guiji.quartz.config.ProxyConfig;
import com.guiji.quartz.formatter.DateFormatterIST;
import com.guiji.quartz.service.ISpiderNewsService;
import com.guiji.quartz.service.RedisService;
import com.guiji.quartz.task.DataSaveUtil;
import com.guiji.quartz.util.DateFormatUtil;
import com.guiji.quartz.util.DelPicUtil;
import lombok.Data;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import us.codecraft.webmagic.Page;
import us.codecraft.webmagic.model.AfterExtractor;
import us.codecraft.webmagic.model.annotation.ExtractBy;
import us.codecraft.webmagic.model.annotation.Formatter;
import us.codecraft.webmagic.model.annotation.TargetUrl;

import java.io.File;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * @program: cms-vue-plus
 * @description:
 * @author: gaoX
 * @create: 2021-10-12 16:31
 **/
public class Gadgets {
	private static final Logger log = LoggerFactory.getLogger(Gadgets.class);

	public void mainMethod() throws Exception {
		HttpResponse response = null;
		try {
			response = HttpRequest.get("https://gadgets.ndtv.com/news")
				.header("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9")
				.header("Accept-Encoding", "gzip, deflate")
				.setHttpProxy(ProxyConfig.url,ProxyConfig.ip)
				.setConnectionTimeout(15000)
				.execute();
		} catch (Exception e) {
			log.error("代理失败1"+e);
			return;
		}
		String resp = response.body();
		List<String> resUrl = getUrl(resp);
		getContent(resUrl);

	}

	private static void getContent(List<String> listContent) {
		for (String url : listContent) {
			log.info("getContent执行到-{},listContent长度-{}", url, listContent.size());
			String doc;
			JXDocument jxDocument;
			try {
				HttpResponse response = HttpRequest.get(url).setHttpProxy(ProxyConfig.url,ProxyConfig.ip).setConnectionTimeout(15000).execute();
				doc = response.body();
				jxDocument = new JXDocument(doc);
			} catch (Exception e) {
				log.info("代理失败2"+e);
				continue;
			}
			String title = null;
			try {
				title = (String) jxDocument.sel("//h1/text()").get(0);
			} catch (Exception e) {
				log.error("标题错误-------------" + url + "-------------------标题错误");
				continue;
			}
			Document document = null;
			String content = null;
			try {
				document = Jsoup.parse(doc);
				content = document.getElementsByClass("content_text").get(0).toString().replaceAll("\n","").replaceAll("iframe.*?}","");
				content = content.replaceAll("<div class=\"content_text row description\">","").replaceAll("\n", "\\$\\$\\$").replaceAll("<div.*?</div>|<ul>.*?</ul>"," ").replaceAll("<.*?>"," ");
			} catch (Exception e) {
				log.error("内容错误-------------" + url + "-------------------内容错误");
				continue;
			}

			List<String> newImgs = null;
			try {
				newImgs = DelPicUtil.delPic(jxDocument.sel("//div[@class='heroimg']//img/@src"));
			} catch (Exception e) {
				log.error("获取图片失败");
				continue;
			}

			Date date = null;
			try {
				date = DateFormatUtil.format1((String) jxDocument.sel("//div[@class='dateline']/text()").get(0));
			} catch (Exception e) {
				log.error("获取时间失败");
				date = new Date();
			}

			if (content.length() > 50000 || content.toString().replaceAll("\\$|[\\s\\p{Zs}]","").length() < 20 || newImgs.size() > 20) {
				continue;
			}
			try {
				int i = RandomUtil.randomInt(1, 3);
				DataSaveUtil.saveData(title, content.toString(), newImgs, url, date, "https://gadgets.ndtv.com/", "Gadgets-IndiaTech-"+i);
				log.info(Thread.currentThread().getName() + "-----------------Gadgets-IndiaTech----------------");
			} catch (Exception e) {
				continue;
			}
		}
	}

	private static List<String> getUrl(String content) {

		List<String> res = new ArrayList<>();
		String pattern = "https://gadgets.ndtv.com/[a-z]{5,20}/news/.*?[0-9]{7}";
		Pattern p = Pattern.compile(pattern);
		Matcher m = p.matcher(content);
		while (m.find()) {
			res.add(m.group());
		}
		return res;
	}

}
